coding problems in python